package com.sicheng.algorithm.array;

public class Find {
    /**
     * 题目描述
     * 给定一个二维数组，其每一行从左到右递增排序，
     * 从上到下也是递增排序。给定一个数，判断这个数是否在该二维数组中。
     * <p>
     * [
     * [1,   4,  7, 11, 15],
     * [2,   5,  8, 12, 19],
     * [3,   6,  9, 16, 22],
     * [10, 13, 14, 17, 24],
     * [18, 21, 23, 26, 30]
     * ]
     * <p>
     * <p>
     * Given target = 5, return true.
     * Given target = 20, return false.
     */
    public static void main(String[] args) {
        int[][] matrix = {
                {1, 4, 7, 11, 15},
                {2, 5, 8, 12, 19},
                {3, 6, 9, 16, 22},
                {10, 13, 14, 17, 24},
                {18, 21, 23, 26, 30}
        };
        boolean isFount = find(10, matrix);
        System.out.println(isFount);
    }

    /**
     * 解题思路
     * 要求时间复杂度 O(M + N)，空间复杂度 O(1)。其中 M 为行数，N 为 列数。
     * <p>
     * 该二维数组中的一个数，小于它的数一定在其左边，大于它的数一定在其下边。
     * 因此，从右上角开始查找，
     * 就可以根据 target 和当前元素的大小关系来快速地缩小查找区间，
     * 每次减少一行或者一列的元素。
     * 当前元素的查找区间为左下角的所有元素。
     *
     * @param target 要寻找的目标
     * @param matrix 要查询的数组
     * @return 是否找到
     */
    public static boolean find(int target, int[][] matrix) {
        if (matrix == null || matrix.length == 0 || matrix[0].length == 0)
            return false;
        int rows = matrix.length, cols = matrix[0].length;
        int r = 0, c = cols - 1; // 从右上角开始
        while (r <= rows - 1 && c >= 0) {
            if (target == matrix[r][c])
                return true;
            else if (target > matrix[r][c])
                r++;
            else
                c--;
        }
        return false;
    }

}

